package com.yunxi.module.system.mapper.permission;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.yunxi.framework.mybatis.core.mapper.BaseMapperX;
import com.yunxi.framework.mybatis.core.query.LambdaQueryWrapperX;
import com.yunxi.module.system.model.permission.UserRoleDO;
import org.apache.ibatis.annotations.Mapper;

import java.util.Collection;
import java.util.List;

/**
 * <p>
 * 用户-角色关联 mapper
 * </p>
 *
 * @author lidy
 * @since 2024-05-07
 */
@Mapper
public interface UserRoleMapper extends BaseMapperX<UserRoleDO> {

    /**
     * 获得拥有多个角色的用户集合
     *
     * @param roleIds 角色集合
     * @return 拥有多个角色的用户集合
     */
    default List<UserRoleDO> selectListByRoleIds(Collection<Long> roleIds) {
        return selectList(UserRoleDO::getRoleId, roleIds);
    }

    /**
     * 获取用户拥有的角色集合
     *
     * @param userId 用户编号
     * @return 角色集合
     */
    default Collection<UserRoleDO> selectListByUserId(Long userId) {
        return selectList(UserRoleDO::getUserId, userId);
    }

    /**
     * 根据用户编号删除用户授权
     *
     * @param userId 用户编号
     */
    default void deleteListByUserId(Long userId) {
        delete(new LambdaQueryWrapperX<UserRoleDO>()
                .eq(UserRoleDO::getUserId, userId));
    }

    /**
     * 根据角色编号删除
     *
     * @param roleId 角色编号
     */
    default void deleteListByRoleId(Long roleId) {
        delete(new LambdaQueryWrapper<UserRoleDO>().eq(UserRoleDO::getRoleId, roleId));
    }

    /**
     * 根据用户编号和角色编号集合删除授权数据
     *
     * @param userId  用户编号
     * @param roleIds 角色编号集合
     */
    default void deleteListByUserIdAndRoleIds(Long userId, Collection<Long> roleIds) {
        delete(new LambdaQueryWrapperX<UserRoleDO>()
                .eq(UserRoleDO::getUserId, userId)
                .in(UserRoleDO::getRoleId, roleIds));
    }
}
